package com.mh.shiro.controller;

import com.github.pagehelper.PageInfo;
import com.mh.shiro.business.entity.User;
import com.mh.shiro.business.service.SysRoleService;
import com.mh.shiro.business.service.SysUserRoleService;
import com.mh.shiro.business.service.SysUserService;
import com.mh.shiro.business.vo.UserConditionVO;
import com.mh.shiro.framework.object.PageResult;
import com.mh.shiro.framework.object.ResponseVO;
import com.mh.shiro.util.PasswordUtil;
import com.mh.shiro.util.ResultUtil;
import org.apache.shiro.authz.annotation.Logical;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.*;
import com.mh.shiro.business.enums.ResponseStatus;


/**
 * @Author: E-mail:
 * @Version: 创建时间：2019年06月21日9:27
 * @Description: <描述>
 */
@RestController
@RequestMapping("/user")
public class RestUserController {

    @Autowired
    private SysUserService userService;
    @Autowired
    private SysUserRoleService userRoleService;
    @Autowired
    private SysRoleService roleService;

    @RequiresPermissions("users")
    @RequestMapping("/list")
    public PageResult list(UserConditionVO vo) {

        PageInfo<User> pageInfo = userService.findPageBreakByCondition(vo);

        return ResultUtil.tablePage(pageInfo);
    }

    @RequiresPermissions("user:add")
    @PostMapping("/add")
    public ResponseVO add(User user) {

        User u = userService.getByUserName(user.getUsername());

        if (u != null) {
            return ResultUtil.error("该用户名[" + user.getUsername() + "]已存在！请更改用户名");
        }
        try {
            user.setPassword(PasswordUtil.encrypt(user.getPassword(), user.getUsername()));
            userService.insert(user);

            return ResultUtil.success("成功");
        } catch (Exception e) {
            e.printStackTrace();
            return ResultUtil.error("error");
        }
    }

    @RequiresPermissions(value = {"user:batchDelete", "user:delete"}, logical = Logical.OR)
    @PostMapping("/remove")
    public ResponseVO remove(Long[] ids) {

        if (null == ids) {
            return ResultUtil.error(500, "请选择至少一条记录");
        }
        for (Long id : ids) {
            userService.removeByPrimaryKey(id);
            userRoleService.removeByUserId(id);
        }
        return ResultUtil.success("成功删除[" + ids.length + "]个用户");
    }

    /**
     * 根据id查找用户角色
     *
     * @param id
     * @return
     */
    @RequiresPermissions("user:edit")
    @PostMapping("/get/{id}")
    public ResponseVO get(@PathVariable Long id) {
        return ResultUtil.success(null, this.userService.getByPrimaryKey(id));
    }

    /**
     * 添加用户角色
     *
     * @param userId
     * @param roleIds
     * @return
     */
    @RequiresPermissions("user:allotRole")
    @PostMapping("/saveUserRoles")
    public ResponseVO saveUserRoles(Long userId, String roleIds) {
        System.out.println(userId + "," + roleIds);

        if (StringUtils.isEmpty(roleIds)) {
            return ResultUtil.error("error");
        }
        userRoleService.addUserRole(userId, roleIds);

        return ResultUtil.success("成功");
    }
    @RequiresPermissions("user:edit")
    @PostMapping("/edit")
    public ResponseVO edit(User user) {
        try {
            userService.updateSelective(user);
        } catch (Exception e) {
            e.printStackTrace();
            return ResultUtil.error("用户修改失败!");
        }
        return ResultUtil.success(ResponseStatus.SUCCESS);
    }


}
